/**
 * This file is mostly a copy of nanopb/examples/network_server/common.h
 */

#include <pb_decode.h>
#include <pb_encode.h>
#include <sys/socket.h>
#include <sys/types.h>

#include "nanopb_stream.h"

static bool write_callback(pb_ostream_t *stream, const uint8_t *buf,
                           size_t count) {
  int fd = (intptr_t)stream->state;
  return send(fd, buf, count, 0) == (ssize_t)count;
}

static bool read_callback(pb_istream_t *stream, uint8_t *buf, size_t count) {
  int fd = (intptr_t)stream->state;
  int result;

  result = recv(fd, buf, count, MSG_WAITALL);

  if (result == 0) {
    stream->bytes_left = 0; /* EOF */
  }

  return result == (int)count;
}

pb_ostream_t pb_ostream_from_socket(int fd) {
  pb_ostream_t stream = {&write_callback, (void *)(intptr_t)fd, SIZE_MAX, 0,
                         NULL};
  return stream;
}

pb_istream_t pb_istream_from_socket(int fd) {
  pb_istream_t stream = {&read_callback, (void *)(intptr_t)fd, SIZE_MAX, NULL};
  return stream;
}
